import os

# 设置根目录路径
# root_path = '../THUCNews/'
# root_path = '../A7_txt_data/'
root_path = '../My_CNews_DataSet/'

# 获取根目录下所有的子目录（分类）
cateList = os.listdir(root_path)
print(cateList)

# 遍历每一个子目录
for cate in cateList:
    # 子目录名称作为文件名的前缀
    prefix = cate

    # 构建子目录的完整路径
    openFileDir = os.path.join(root_path, cate)

    # 获取子目录下所有文件的列表
    fileList = os.listdir(openFileDir)

    # 遍历文件列表，并对每个文件进行重命名
    for index, file in enumerate(fileList):
        # 构建新的文件名：前缀 + 文件索引
        new_file_name = f"{prefix}_{index}" + '.txt'
        # new_file_name = f"HH_{index}" + '.txt'

        # 构建旧文件的完整路径
        old_file_path = os.path.join(openFileDir, file)

        # 构建新文件的完整路径
        new_file_path = os.path.join(openFileDir, new_file_name)

        # 检查新文件名是否已存在
        if os.path.exists(new_file_path):
            print(f"警告：文件'{new_file_path}'已存在，跳过重命名操作。")
            continue

        # 执行重命名操作
        os.rename(old_file_path, new_file_path)

        # 打印重命名操作的结果
        print(f"文件'{old_file_path}'已重命名为'{new_file_path}'")
